import torch

print("PyTorch版本:", torch.__version__)
print("CUDA是否可用:", torch.cuda.is_available())
if torch.cuda.is_available():
    print("CUDA版本:", torch.version.cuda)
    print("GPU设备数量:", torch.cuda.device_count())
    print("当前GPU设备名称:", torch.cuda.get_device_name(0))
else:
    print("CUDA不可用，请检查您的GPU驱动和CUDA安装")
